import argparse
import torch

parser = argparse.ArgumentParser()

parser.add_argument('--train_file', default='data/train.txt')
parser.add_argument('--dev_file', default='data/dev.txt')
parser.add_argument('--test_file', default='data/text.txt')

parser.add_argument('--bert_model', default='bert-base-uncased')
parser.add_argument('--do_lower_case', action='store_true')
parser.add_argument('--checkpoint', default=1, type=int)
parser.add_argument('--save_dir', default='', type=str)

parser.add_argument('--model_type', default='model', type=str)
parser.add_argument('--pair', action='store_true')
parser.add_argument('--seed', default=-1, type=int)

#TODO 常改动参数
parser.add_argument('--do_train', action='store_true')
parser.add_argument('--learning_rate', default=1e-5, type=float)
parser.add_argument('--adam_epsilon', default=1e-8, type=float)

parser.add_argument('--gpu', default=0, type=int)  #卡号
parser.add_argument('--epochs', default=20, type=int)  #迭代次数
parser.add_argument('--batch_size', default=32, type=int)
parser.add_argument('--max_length', default=0, type=int)

parser.add_argument('--s1_length', default=50, type=int)  #这两个length是干啥的？
parser.add_argument('--s2_length', default=100, type=int)

parser.add_argument("--warmup_steps", default=0, type=int)
parser.add_argument('--fp16', action='store_true')
parser.add_argument('--fptype', default='01')

args = parser.parse_args()
#print(args)
device = torch.device('cuda:' + str(args.gpu) if torch.cuda.is_available() else 'cpu')
args.device = device